// 版权所有2011 Go作者。版权所有。
// 此源代码的使用受BSD样式的约束
// 可以在许可证文件中找到的许可证。

package bzip2

import (
	"bufio"
	"io"
)

// bitReader包装io.Reader并提供读取值的功能，
// 一点一点地，从中。它的Read*方法不会返回通常的错误
// 因为错误处理是冗长的。相反，任何错误都会被保留并可以
// 事后检查。
type bitReader struct {
	r    io.ByteReader
	n    uint64
	bits uint
	err  error
}

// newBitReader返回从r读取的新位读取器。如果r不是
// 已经是io.ByteReader，将通过bufio.Reader进行转换。
func newBitReader(r io.Reader) bitReader {
	byter, ok := r.(io.ByteReader)
	if !ok {
		byter = bufio.NewReader(r)
	}
	return bitReader{r: byter}
}

// ReadBits64读取给定数量的位，并在
// uint64的最低有效部分。如果发生错误，它将返回0
// 并且可以通过调用Err（）获得错误。
func (br *bitReader) ReadBits64(bits uint) (n uint64) {
	for bits > br.bits {
		b, err := br.r.ReadByte()
		if err == io.EOF {
			err = io.ErrUnexpectedEOF
		}
		if err != nil {
			br.err = err
			return 0
		}
		br.n <<= 8
		br.n |= uint64(b)
		br.bits += 8
	}

	// br.n如下所示（假设br.bits=14，bits=6）：
	// 比特：111111
	// 5432109876543210
	// None
	// （6位，所需输出）
	// |-----|
	// V V
	// 0101101101001110
	// ^            ^
	// |------------|
	// br.bits（num有效位）
	// None
	// 下一行将所需位右移到
	// 最不重要的地方和遮罩上面的任何东西。
	n = (br.n >> (br.bits - bits)) & ((1 << bits) - 1)
	br.bits -= bits
	return
}

func (br *bitReader) ReadBits(bits uint) (n int) {
	n64 := br.ReadBits64(bits)
	return int(n64)
}

func (br *bitReader) ReadBit() bool {
	n := br.ReadBits(1)
	return n != 0
}

func (br *bitReader) Err() error {
	return br.err
}
